AI SDK Kotlin/ai.torad.aisdk.providers/MockLanguageModel

MockLanguageModel

class MockLanguageModel(val modelId: String = "mock/test", val provider: String = "mock", responses: List<ScriptedResponse>) : LanguageModel

Test stub for a LanguageModel. Drives the agent loop deterministically by replaying a list of ScriptedResponses — one per stream call.

Lets tests verify the loop, the tool dispatch, the lifecycle hooks, the stop conditions, and the message-parts conversion without spinning up a real model.

Constructors

MockLanguageModel

constructor(modelId: String = "mock/test", provider: String = "mock", responses: List<ScriptedResponse>)

Properties

modelId

open override val modelId: String

Stable identifier for telemetry / routing ("litertlm/gemma-3-1b", "openai/gpt-5").

provider

open override val provider: String

Provider tag for the model — "openai", "anthropic", "litert-lm", etc. Mirrors v6's LanguageModelV3.provider (per historical parity gap #40). Lets routing layers (a fallback chain, a tier picker) make provider-aware decisions without parsing modelId.

supportedUrls

open val supportedUrls: Map<String, List<String>>

Map of URL regex → media-tag list describing which URL shapes the provider accepts inline (so the agent doesn't have to download + base64-encode them itself). Mirrors v6's LanguageModelV3.supportedUrls. Empty default — most on-device providers don't accept URLs and fall back to base64.

Functions

generate

open suspend override fun generate(params: LanguageModelCallParams): LanguageModelResult

One-shot completion.

stream

open override fun stream(params: LanguageModelCallParams): Flow<StreamEvent>

Streaming completion. Cold until collected, then drives one upstream call per collection.

streamResult

open fun streamResult(params: LanguageModelCallParams): LanguageModelStreamResult

Streaming completion plus metadata available before stream collection. Implementations must keep the returned stream cold, and must not flowOn/emit from a foreign context — the collection context is the caller's choice. Providers can override this to expose request bodies and response headers while preserving the v6 doStream result shape.